op {
  graph_op_name: "TPUReshardVariables"
  visibility: HIDDEN
  summary: "Op that reshards on-device TPU variables to specified state."
  description: <<END
Op that reshards on-device TPU variables to specified state. Internal use only.

The sharding state is represented as the key of the compilation that generated
the sharding/unsharding programs along with the main program. new_format_key
specifies the desired state, and format_state_var is the current state of the
variables.
END
}
